package cc.owoo.godpen.analysis.html;

/**
 * 元素集合
 * Created by nimensei
 * 2022-05-21 上午 01:58
 **/
public class ElementChildList extends ElementList {
    private final ElementMap idMap = new ElementMap(ElementMap.ElementType.ELEMENT);// id子元素映射表

    @Override
    void add(Element element) {
        String id = element.getId();
        if (id != null)
            idMap.add(id, element);
        super.add(element);
    }

    /**
     * 根据ID获取元素
     *
     * @param id 元素ID
     * @return 元素对象
     */
    @Override
    public Element get(String id) {
        return idMap.get(id).get(0);
    }
}
